package utils

import (
	"mime/multipart"
	"path"
)

// CheckImageExt 检测图片后缀名是否合法
func CheckImageExt(file *multipart.FileHeader) bool {
	extName := path.Ext(file.Filename)
	allowExtMap := map[string]bool{ // 设置合法后缀名
		".jpg":  true,
		".png":  true,
		".gif":  true,
		".jpeg": true,
		".md":   true,
	}
	if _, ok := allowExtMap[extName]; !ok { // 判断后缀名是否在符合类型列表中
		return false
	}
	return true
}
